package com.saturday.web.common;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletRequestWrapper;
import java.util.HashMap;
import java.util.Map;

public class ParameterRequestWrapper extends HttpServletRequestWrapper {

    /**
     * 初始化params,作为getParameter方法获取的参数列表,内首先包含req中的paramMap,再修改
     */
    private Map<String, String[]> params = new HashMap<String, String[]>();

    /**
     * 重载构造器,并调用父类构造器,将参数写入子类params
     *
     * @param request
     */
    public ParameterRequestWrapper(HttpServletRequest request) {
        // 将request交给父类，以便于调用对应方法的时候，将其输出，其实父亲类的实现方式和第一种new的方式类似
        super(request);
        //将参数表，赋予给当前的Map以便于持有request中的参数
        this.params.putAll(request.getParameterMap());
    }

    /**
     * 重写getParameter，代表参数从当前类中的map获取
     */
    @Override
    public String getParameter(String name) {
        String[] values = params.get(name);
        if (values == null || values.length == 0) {
            return null;
        }
        return values[0];
    }

    /**
     * 获取子类param缓存的方法
     */
    public Map<String, String[]> getParamsMap() {
        return this.params;
    }

    /**
     * 写入一个map的方法
     *
     * @param map
     */
    public void addParamsMap(Map<String, Object> map) {
        for (Map.Entry<String, Object> entry : map.entrySet()) {
            addParam(entry.getKey(), entry.getValue());
        }
    }

    /**
     * 写入一个参数的方法
     *
     * @param name
     * @param obj
     */
    public void addParam(String name, Object obj) {
        if (obj != null) {
            if (obj instanceof String[]) {
                params.put(name, (String[]) obj);
            } else if (obj instanceof String) {
                params.put(name, new String[]{(String) obj});
            } else {
                params.put(name, new String[]{String.valueOf(obj)});
            }
        }
    }

}